import os
import torch


class Config:
    root_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
    model_path = '/data0/jianyu10/PTM/huggingface_model_cache/chinese-roberta-wwm-ext'
    # model_cache = '/data0/jianyu10/PTM/huggingface_model_cache/chinese-roberta-wwm-ext'
    # model_path = 'D:/PTM/chinse-roberta-wwm-ext'

    dropout_prob = 0.4
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

    replace_token = "[X]"
    mask_token = "[MASK]"
    mask_ids = 103
    prompt_templates = ['"{}"，它的意思是[MASK]。'.format(replace_token), '"{}"，这句话的意思是[MASK]。'.format(replace_token)]
    dropout = 0.4
    train_path = root_path + '/data/train.csv'
    valid_path = root_path + '/data/devdata.csv'
    log_dir = root_path + '/Log/'
    batch_size = 8
    lr = 5e-6
    num_epochs = 300
    require_improvement = 60

    save_path = root_path + '/CheckPoint/'
